import random
import time
import os

"""
    本模块默认保存位置: 当前文件夹下的 output 
"""

# 默认输出路径
output_dir = "./output/"

def random_filename():
    """
        随机文件名
        :return 返回文件名
    """
    filename = str(time.time())
    return filename.replace(".","")

def check():
    """ output 文件夹是否被创建 """
    if os.path.exists(output_dir) == False:
        os.system("mkdir output")


def file(saved_text, filename="", filetype=".txt"):
    """
        保存文件, 默认保存为记事本格式
        :param saved_text 要保存的文本
        :param filename 文件名
        :param filetype 文件类型
    """
    check()

    if filename == "":
        filename = random_filename()    

    # 保存文件路径拼接
    save_path = output_dir + filename + filetype

    with open(save_path, 'w', encoding="utf-8") as f:
        f.write(saved_text)

    # 返回保存路径
    return save_path


def html(saved_text, filename=""):
    """
    保存网页文件到本地
    :param saved_text 要保存的网页文本
    """ 
    return file(saved_text, filename, filetype=".html")
    